package org.zfes.snowy.base.dao.mybatis;

import java.io.ByteArrayInputStream;
import java.io.UnsupportedEncodingException;
import java.sql.Blob;
import java.sql.CallableStatement;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;

import org.apache.ibatis.type.BaseTypeHandler;
import org.apache.ibatis.type.JdbcType;
import org.zfes.snowy.core.util.ZObjectUtil;


public class BaseBlobTypeHandler extends BaseTypeHandler<String> {
	private static final String DEFAULT_CHARSET = "utf-8";

	@Override  
    public void setNonNullParameter(PreparedStatement ps, int i,  
            String parameter, JdbcType jdbcType) throws SQLException {  
        ByteArrayInputStream bis;  
        try {  
            bis = new ByteArrayInputStream(parameter.getBytes(DEFAULT_CHARSET));  
        } catch (UnsupportedEncodingException e) {  
            throw new RuntimeException("Blob Encoding Error!");  
        }     
        ps.setBinaryStream(i, bis, parameter.length());  
    }

	@Override
	public String getNullableResult(ResultSet rs, String columnName) throws SQLException {
		Blob blob = rs.getBlob(columnName);
		return getBolbValue(blob);
	}

	@Override
	public String getNullableResult(CallableStatement cs, int columnIndex) throws SQLException {
		Blob blob = cs.getBlob(columnIndex);
		return getBolbValue(blob);
	}

	@Override
	public String getNullableResult(ResultSet rs, int columnIndex) throws SQLException {
		Blob blob = rs.getBlob(columnIndex);
		return getBolbValue(blob);
	}
	private String getBolbValue(Blob blob) throws SQLException{
		byte[] returnValue = null;
		if (!ZObjectUtil.isEmpty(blob)) {
			returnValue = blob.getBytes(1, (int) blob.length());
		}else{
			return "";
		}
		try {
			if(!ZObjectUtil.isEmpty(returnValue)){
				return new String(returnValue, DEFAULT_CHARSET);
			}else{
				return "";
			}
		} catch (UnsupportedEncodingException e) {
			throw new RuntimeException("Blob Encoding Error!");
		}
	}
}
